import os

os.chdir(r'D:\gitee_codes\transformer-learning\transformers-master\examples\question-answering')

cmd = 'python run_squad.py \
	    --model_type bert \
	    --model_name_or_path D:/BaiduNetdiskDownload/预训练模型/huggingface/bert-base-uncased \
        --do_train \
	    --do_eval \
	    --do_lower_case \
	    --train_file D:/BaiduNetdiskDownload/预训练模型/datasets/SQUAD/SQUAD1.1/train-v1.1.json \
	    --predict_file D:/BaiduNetdiskDownload/预训练模型/datasets/SQUAD/SQUAD1.1/dev-v1.1.json \
	    --per_gpu_train_batch_size 2 \
	    --learning_rate 3e-5 \
	    --num_train_epoch 2.0 \
	    --max_seq_length 384 \
	    --doc_stride 128 \
	    --output_dir D:/gitee_codes/transformer-learning/SQUAD_OUTPUT \
	    --overwrite_output_dir\
	    --gradient_accumulation_steps 3\
'
os.system(cmd)
